package com.demo.config.desensitizationConfig;

import cn.hutool.core.util.ObjectUtil;
import com.fasterxml.jackson.core.JsonGenerator;
import com.fasterxml.jackson.databind.BeanProperty;
import com.fasterxml.jackson.databind.JsonMappingException;
import com.fasterxml.jackson.databind.JsonSerializer;
import com.fasterxml.jackson.databind.SerializerProvider;
import com.fasterxml.jackson.databind.ser.ContextualSerializer;

import java.io.IOException;

/**
 * 自定义的脱敏JSON序列化器 在该序列化器中找到带有该注解的字段，根据注解中指定的脱敏策略，在序列化过程中将数据进行脱敏处理并输出到 JSON 中。
 * 平时接口返回的数据结构直接都是交给默认的序列化器把对象转成json的。
 * 实现脱敏的本质就是在这前面添加了一段逻辑，找到带有脱敏注解的属性然后拿到注解指定的脱敏策略实例化这个脱敏策略枚举类，
 * 应用对应的脱敏方法处理需要脱敏字段，得到脱敏后的值交给json生成器得到最终脱敏后的json
 *
 * @author wgr
 */
public class DesensitizationJsonSerializable extends JsonSerializer<String> implements ContextualSerializer {

	private DesensitizationStrategyEnum desensitizationStrategy; // 脱敏策略

	@Override
	public void serialize(String s, JsonGenerator jsonGenerator, SerializerProvider serializerProvider)
			throws IOException {
		// 将字符串按照设定的脱敏策略进行脱敏处理后序列化到 JSON 中
		jsonGenerator.writeString(desensitizationStrategy.getDesensitization().apply(s));
	}

	@Override
	public JsonSerializer<?> createContextual(SerializerProvider serializerProvider, BeanProperty beanProperty)
			throws JsonMappingException {
		// 获取属性上的 Desensitization 注解
		Desensitization annotation = beanProperty.getAnnotation(Desensitization.class);

		// 判断注解不为空且属性类型为 String
		if (ObjectUtil.isNotNull(annotation) && ObjectUtil.equals(String.class, beanProperty.getType().getRawClass())) {
			this.desensitizationStrategy = annotation.desensitizationStrategy(); // 设置脱敏策略
			return this;
		}

		// 返回默认的序列化器
		return serializerProvider.findValueSerializer(beanProperty.getType(), beanProperty);
	}

}